from app import create_app, db
from app.models import User, Couple, Quote
from werkzeug.security import generate_password_hash

app = create_app()


def init_db():
    with app.app_context():
        # 先创建所有表
        db.create_all()

        # 检查是否已经有数据
        if User.query.first() is not None:
            print("数据库已经初始化过了")
            return

        # 创建管理员用户
        admin = User(
            username='admin',
            email='575732022@qq.com',
            password_hash=generate_password_hash('123456'),
            is_admin=True
        )
        db.session.add(admin)

        # 创建一些夫妻数据
        couples = [
            {
                'season': 4,
                'husband_name': '杨子 作者QQ:575732022',
                'husband_age': 48,
                'husband_occupation': '演员、出品人、制片人',
                'wife_name': '黄圣依',
                'wife_age': 40,
                'wife_occupation': '演员、歌手、出品人、商人',
                'marriage_years': 17,
                'story': '黄圣依和杨子结婚17年，他们的婚姻备受关注。在节目中，黄圣依想看杨子离开工作能否正常生活，杨子则表示参加节目很痛苦，两人的相处模式和婚姻状态在节目中得以展现，如婚姻体检成绩不理想等，他们面临着各种婚姻中的挑战和问题，在旅途中不断探索和面对彼此的关系。',
                'likes_count': 0,
                'husband_avatar':'/static/images/401.webp',
                'wife_avatar':'/static/images/400.webp'
            },
            {
                'season': 4,
                'husband_name': '李行亮',
                'husband_age': 40,
                'husband_occupation': '歌手',
                'wife_name': '麦琳',
                'wife_age': 38,
                'wife_occupation': '家庭主妇',
                'marriage_years': 10,
                'story': '麦琳非常节约，掌握家中财政大权，李行亮几乎没有零花钱且花钱需向麦琳打报告，他们在婚姻中对彼此的角色认知存在差异，在相处过程中也面临着如经济管理等方面的矛盾和困惑，在节目旅程中逐渐重新审视彼此的关系。',
                'likes_count': 0,
                'husband_avatar':'/static/images/411.webp',
                'wife_avatar':'/static/images/410.webp'
            },
            {
                'season': 4,
                'husband_name': '刘爽',
                'husband_age': 35,
                'husband_occupation': '自媒体人',
                'wife_name': '葛夕',
                'wife_age': 33,
                'wife_occupation': '自媒体人',
                'marriage_years': 10,
                'story': '刘爽和葛夕结婚初期，刘爽一吵架就下跪道歉，现在葛夕是非分明，认为下跪不能解决问题，他们在婚姻中不断成长和改变，在节目中的旅行里进一步面对和处理彼此之间的关系问题，展现出婚姻中的变化和发展。',
                'likes_count': 0,
                'husband_avatar':'/static/images/421.webp',
                'wife_avatar':'/static/images/420.webp'
            },
            {
                'season': 3,
                'husband_name': '刘毅',
                'husband_age': 40,
                'husband_occupation': '自由职业者',
                'wife_name': '傅首尔',
                'wife_age': 39,
                'wife_occupation': '女辩手、作家、主持人',
                'marriage_years': 15,
                'story': '傅首尔和刘毅结婚15年，从校园到婚纱，曾是令人羡慕的一对。但随着傅首尔事业发展，两人节奏出现差距，工作忙碌的傅首尔与全职在家的刘毅沟通和爱减少，生日卡片内容也引发争议，两人在节目中探讨婚姻中的问题，包括爱情的缺失、付出方式的差异等，在旅途中不断重新审视彼此关系，最终未选择下车，选择了放手。',
                'likes_count': 0,
                'husband_avatar':'/static/images/301.webp',
                'wife_avatar':'/static/images/300.webp'
            },
            {
                'season': 3,
                'husband_name': '张硕',
                'husband_age': 30,
                'husband_occupation': '广告制片人',
                'wife_name': '王睡睡',
                'wife_age': 30,
                'wife_occupation': '美妆博主',
                'marriage_years': 3,
                'story': '王睡睡和张硕从高中开始恋爱，经历多年，婚礼上的“婚闹”事件成为他们感情的阴影，导致至今未领证，婚后生活中存在诸多问题，如张硕在王睡睡发烧时的不当处理、生活习惯差异等，在节目中两人矛盾不断爆发，王睡睡多次对张硕的行为表示不满，最终王睡睡选择不下车，两人分手。',
                'likes_count': 0,
                'husband_avatar':'/static/images/321.webp',
                'wife_avatar':'/static/images/320.webp'
            },
            {
                'season': 3,
                'husband_name': '纪焕博',
                'husband_age': 40,
                'husband_occupation': '演员、模特',
                'wife_name': '王诗晴',
                'wife_age': 30,
                'wife_occupation': '模特',
                'marriage_years': 12,
                'story': '纪焕博和王诗晴的婚姻中，纪焕博因专业态度对王诗晴的穿搭等方面控制欲较强，王诗晴起初享受后因自我意识觉醒开始反抗，两人在节目中多次发生争吵，如因王诗晴给男性朋友回复表情引发矛盾等，但在旅程中也有相互理解和改变的时刻，最终抉择时两人都下车，选择继续这段婚姻。',
                'likes_count': 0,
                'husband_avatar':'/static/images/311.webp',
                'wife_avatar':'/static/images/310.webp'
            },
            {
                'season': 2,
                'husband_name': '艾威',
                'husband_age': 60,
                'husband_occupation': '演员',
                'wife_name': '陈美玲',
                'wife_age': 57,
                'wife_occupation': '手表品牌店店长',
                'marriage_years': 18,
                'story': '陈美玲和艾威相识31年，结婚18年，在六十岁选择离婚。艾威曾散尽家财为陈美玲治病，但因陈美玲不顾身体健康通宵打麻将提出离婚，后在旅行中两人关系逐渐变化，如艾威在陈美玲生日时准备惊喜，陈美玲生病时悉心照料，最终两人都下车，和好如初。',
                'likes_count': 0,
                'husband_avatar':'/static/images/201.webp',
                'wife_avatar':'/static/images/200.webp'
            },
            {
                'season': 2,
                'husband_name': '卢歌',
                'husband_age': 35,
                'husband_occupation': '舞蹈演员',
                'wife_name': '苏诗丁',
                'wife_age': 33,
                'wife_occupation': '歌手、声乐教师',
                'marriage_years': 3,
                'story': '苏诗丁和卢歌因婚后长期异地离婚，在节目中两人存在诸多问题，如缺乏深度沟通，对彼此需求认知不同，在旅行中多次发生争吵和矛盾，如卢歌对苏诗丁讲述童年的反应过激，夫妻画像环节两人对彼此的认知差异巨大等，最终苏诗丁做出下车又离开的选择，与卢歌遗憾错过。',
                'likes_count': 0,
                'husband_avatar':'/static/images/211.webp',
                'wife_avatar':'/static/images/210.webp'
            },
            {
                'season': 2,
                'husband_name': '宋宁峰',
                'husband_age': 40,
                'husband_occupation': '演员、模特',
                'wife_name': '张婉婷',
                'wife_age': 33,
                'wife_occupation': '职业经纪人',
                'marriage_years': 3,
                'story': '张婉婷和宋宁峰恋爱一个月闪婚闪孕，婚后存在诸多问题，如张婉婷强势、宋宁峰沉默，常引发争吵，张婉婷多次提离婚，两人在旅行中矛盾不断，如张婉婷限制宋宁峰发言等，但也有互相理解和温暖的时刻，最终宋宁峰下车求婚，张婉婷下车，两人继续婚姻。',
                'likes_count': 0,
                'husband_avatar':'/static/images/221.webp',
                'wife_avatar':'/static/images/220.webp'
            },
            {
                'season': 1,
                'husband_name': '章贺',
                'husband_age': 40,
                'husband_occupation': '演员',
                'wife_name': '郭柯宇',
                'wife_age': 45,
                'wife_occupation': '演员',
                'marriage_years': 10,
                'story': '章贺和郭柯宇相识于共同主演的电影后闪婚，婚姻里彼此感到孤独，矛盾逐渐深化最终离婚。在节目旅行中，两人从最初的疏远到逐渐走近，有了新的默契，如一起骑车时亲密互动等，但在交流中仍存在问题，章贺对过去十年的关系存在心结，最终章贺提前下车希望郭柯宇也下车，郭柯宇则在纠结后未下车，两人遗憾错过。',
                'likes_count': 0,
                'husband_avatar':'/static/images/101.webp',
                'wife_avatar':'/static/images/100.webp'
            },
            {
                'season': 1,
                'husband_name': '魏巍',
                'husband_age': 38,
                'husband_occupation': '主持人、演员',
                'wife_name': '佟晨洁',
                'wife_age': 39,
                'wife_occupation': '模特、主持人、演员',
                'marriage_years': 7,
                'story': '魏巍和佟晨洁结婚7年，未育孩子及魏巍爱喝酒等问题导致夫妻矛盾，魏巍希望有孩子且认为自己不做饭是为让佟晨洁感受被需要，佟晨洁则担心丧偶式育娃。旅行中两人多次产生冲突，如魏巍因佟晨洁对其做饭问题的态度而情绪失控，也有温馨时刻，如魏巍教佟晨洁骑单车，最终抉择时魏巍下车，佟晨洁虽不确定但表示能接受结果，看到魏巍追车后两人和好。',
                'likes_count': 0,
                'husband_avatar':'/static/images/111.webp',
                'wife_avatar':'/static/images/110.webp'
            },
            {
                'season': 1,
                'husband_name': '王秋雨',
                'husband_age': 50,
                'husband_occupation': '编剧',
                'wife_name': '朱雅琼',
                'wife_age': 35,
                'wife_occupation': '歌手',
                'marriage_years': 6,
                'story': '王秋雨和朱雅琼年龄相差10岁，价值观不同，朱雅琼在婚姻中未得到尊重和仪式感，虽王秋雨对朱雅琼有感情但不重视其情感需求。旅行中两人矛盾不断，如朱雅琼因王秋雨在游戏中表示讨厌她唱歌而崩溃，也有缓和时刻，如王秋雨帮朱雅琼找回丢失相机，最终两人拥抱祝福，彼此下车。',
                'likes_count': 0,
                'husband_avatar':'/static/images/121.webp',
                'wife_avatar':'/static/images/120.webp'
            }
        ]

        for couple_data in couples:
            couple = Couple(**couple_data)
            db.session.add(couple)

        # 创建一些名言
        quotes = [
            {
                'content': '再见爱人Web 基于Python Flask框架 - 原创作品 - 灵感来源芒果TV《再见爱人》综艺',
                'author': 'QQ：575732022',
                'is_active': True
            },
            {
                'content': '婚姻不是爱情的坟墓，而是爱情的考验场',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '不是所有的感情都会有结果，但每一段感情都会有收获',
                'author': '佚名',
                'is_active': True
            }, {
                'content': '分手不是结束，而是重新开始寻找真爱的起点。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '当爱情走到尽头，放手也是一种温柔。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手了，就别再回头，因为回头也看不到曾经的美好。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情离去时，不必强留，留下的或许只是伤痛。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手如同剪掉的指甲，再长出来也不是原来的模样。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '一段感情的结束，是为了给更合适的爱腾出位置。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手是人生的一堂课，教会我们如何更好地去爱下一个人。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情散场时，微笑告别，好过哭着挽留。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '结束一段感情，就像打破一面镜子，即便拼凑起来也有了裂痕。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手并非失败，而是勇敢面对不适合的爱情。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情离去后，回忆是唯一留下的礼物，或甜或苦。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '当分手来临，坦然接受，是对彼此最后的尊重。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '一段感情的终结，往往是新的自我成长的开端。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手了，别让怨恨蒙蔽双眼，记得曾经的爱就好。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情如烟花，绚烂过后的分手，是回归平淡的必然。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '结束的感情，如同过期的车票，无法再登上那趟爱的列车。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手是割舍，也是解脱，从一段不合适的爱中挣脱出来。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情走了，留下的空白，要用时间和新的经历去填补。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '面对分手，坚强是最好的伪装，眼泪留给自己就好。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '一段感情结束，不是世界末日，而是新的人生篇章开启。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手让我们明白，有些爱，只能陪你走一段路。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情离去的背影，虽有些落寞，但也是新的开始的信号。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '结束感情时，不要追问为什么，有时候答案并不重要。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手如同换季，脱下旧的情感外衣，迎接新的可能。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '一段感情的结束，往往伴随着成长的阵痛。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情走了，带走了甜蜜，留下的苦涩也是人生滋味。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '当分手成为定局，释怀是对自己最大的宽容。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '结束一段感情，就像告别一场美梦，醒来虽有失落但要继续前行。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手是人生路上的一个岔口，选择不同方向，遇见不同风景。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情离去，不必执着，前方或许有更美的风景在等你。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '一段感情结束，我们学会了在伤痛中坚强，在失去中成长。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手了，让过去的爱成为回忆，未来还有无限可能。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情如流水，分手时，只能任其逝去，无法挽留。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '结束感情，是给彼此一个重新寻找幸福的机会。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手是生活的调味剂，虽苦但能让我们更懂得珍惜。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '一段感情的终结，意味着新的故事即将开始。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情走了，回忆却如影随形，不过时间会慢慢淡化它。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '当分手来临，勇敢面对，因为逃避解决不了任何问题。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '结束一段感情，就像合上一本读过的书，虽有不舍但要开启新的阅读。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手是成长的代价，我们用失去一段感情，换来更成熟的自己。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情离去后，我们要学会在孤单中寻找内心的宁静。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '一段感情结束，不是终点，而是通往更好爱情的起点。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手了，把遗憾留在过去，带着希望走向未来。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情如花朵，凋谢后的分手，是自然的轮回。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '结束感情，是为了让彼此不再互相折磨，各自安好。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手是人生的一道坎，跨过去，会看到不一样的风景。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '一段感情的结束，让我们懂得了爱的珍贵与脆弱。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情走了，我们要用微笑去掩盖心中的伤痛，继续生活。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '当分手成为必然，接受现实是最明智的选择。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '结束一段感情，就像拔掉一颗蛀牙，虽痛但长痛不如短痛。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手是重新审视自己的契机，看看在感情中缺失了什么。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情离去，不必过于悲伤，因为这只是人生的一段插曲。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '一段感情结束，我们要学会放下过去的包袱，轻装上阵。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手了，愿彼此都能在没有对方的日子里，过得幸福。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情如流星，璀璨却短暂，分手是它划过夜空后的寂静。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '结束感情，是给彼此的心灵松绑，让它们自由飞翔。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手是人生的一次洗礼，洗去旧的情感尘埃，迎接新的阳光。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '一段感情的终结，让我们明白，不是所有的爱都能永恒。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情走了，我们要学会在寂寞中品味生活的真谛。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '当分手来临，不要让自己陷入绝望，因为希望总会在转角处。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '结束一段感情，就像走出一场迷雾，虽有些迷茫但会渐渐清晰。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手是成长的阶梯，每一次攀登，都让我们变得更加坚强。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情离去后，我们要学会用新的兴趣爱好填充生活的空白。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '一段感情结束，不是末日来临，而是重生的开始。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手了，希望彼此都能成为更好的人，即使不再相伴。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情如梦境，醒来后的分手，是回归现实的无奈。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '结束感情，是为了让彼此能够重新定义自己的幸福。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手是人生的一场修行，修的是面对失去的坦然与豁达。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '一段感情的终结，让我们学会了如何在伤痛中自愈。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情走了，我们要学会在孤独中寻找自我，重新认识自己。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '当分手成为现实，调整心态，迎接新的生活才是关键。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '结束一段感情，就像关闭一扇旧门，同时打开一扇新门。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手是重新出发的号角，吹响它，迈向新的人生旅程。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情离去后，我们要学会用乐观的心态看待未来的生活。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '一段感情结束，不是结束一切，而是开启新的可能。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手了，愿彼此都能在人生的道路上，找到属于自己的幸福归宿。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情如烟火般绚烂，分手时的落寞，是绚烂过后的平淡。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '结束感情，是为了让彼此不再被过去束缚，自由前行。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '分手是人生的一个转折，转身后，或许会遇见更好的爱情。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '一段感情的终结，让我们懂得了放手也是一种爱。',
                'author': '佚名',
                'is_active': True
            },
            {
                'content': '爱情走了，我们要学会在寂静中聆听内心的声音，重新规划人生。',
                'author': '佚名',
                'is_active': True
            }
        ]

        for quote_data in quotes:
            quote = Quote(**quote_data)
            db.session.add(quote)

        try:
            db.session.commit()
            print("数据库初始化成功！")
        except Exception as e:
            db.session.rollback()
            print(f"初始化失败: {str(e)}")


if __name__ == '__main__':
    init_db()
